import pandas as pd

from base import getEnvPath

path = getEnvPath()
# 设置数据文件路径
data_file = path + "data/output_dir/train_data.csv"
# 读取CSV文件（替换成你的文件路径）
data = pd.read_csv(data_file)

# 拆分特征（X）和标签（y）
X_train = data.drop(columns=["Label"])  # 去掉Label列，保留所有特征
y_train = data["Label"]  # 只保留Label列

# 保存拆分后的文件
X_train.to_csv(path + "data/output_dir/X_train.csv", index=False)
y_train.to_csv(path + "data/output_dir/Y_train.csv", index=False)

print("拆分完成！")
